Solution.py

fromqueueimportSimpleQueue


classSolution:
defwatchedVideosByFriends(self, watchedVideos: List[List[str]], friends: List[List[int]], id: int, level: int) ->List[str]:
levels= [100] *len(friends)
levels[id] =0
queue=SimpleQueue()
queue.put((id, 0))
count= {}
ret= []

whilenotqueue.empty():
i, k=queue.get()

ifk>level:
break

forfriendinfriends[i]:
iflevels[friend] >k+1:
levels[friend] =k+1
queue.put((friend, k+1))

foriinrange(len(friends)):
iflevels[i] ==level:
forvideoinwatchedVideos[i]:
ifvideonotincount:
count[video] =0
ret.append(video)
count[video] +=1

returnsorted(ret, key=lambdav: (count[v], v))